
from ultralytics import YOLO


if __name__ == '__main__':
    model = YOLO(r'../../model/yolo11n.yaml')
    model.train(data=r'data.yaml',
                imgsz=640,
                epochs=100,
                single_cls=False,  # 多类别设置False
                batch=16,
                workers=8,
                device=[0,1]
                )
